package com.lss.b_doublePointer;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @author lss
 * @description 给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j],
 * nums[k]] 满足 i != j、i != k 且 j != k ，
 * 同时还满足 nums[i] + nums[j] + nums[k] == 0 。请你返回所有和为 0 且不重复的三元组。
 * @createDate 2024/11/29-11:01
 */
public class lc_15 {
    public static void main(String[] args) {

    }

    /**
     * 原来的解法思路存在问题
     * 采用双指针 and 跳过重复元素的方法来解决
     *
     * @param nums
     * @return
     */
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        Arrays.sort(nums);
        for (int i = 0; i < nums.length - 2; i++) {
            //跳过重复元素 todo 处理之前的元素，不应该处理之后的元素
            // 这个应该是代码执行顺序
            if (i > 0 && nums[i] == nums[i - 1]) continue;
            //开启左右指针
            int left = i + 1;
            int right = nums.length - 1;
            while (left < right) {
                int sum = nums[i] + nums[left] + nums[right];

                if (sum == 0) {
                    res.add(Arrays.asList(nums[i], nums[left], nums[right]));
                    // 缺失判断重复元素逻辑
                    while (left < right && nums[left] == nums[left + 1]) left++;
                    while (left < right && nums[right] == nums[right - 1]) right--;
                    // 不要忘了移动指针
                    left++;
                    right--;
                }
                // 因为进行过排序处理
                else if (sum < 0) {
                    left++;
                } else {
                    right--;
                }
            }
        }

        return res;
    }
}
